package com.c2b.algorithm.leetcode.base;

import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

/**
 * <a href="https://leetcode.cn/problems/keys-and-rooms/">钥匙和房间(Keys and Rooms)</a>
 * <p>有 n 个房间，房间按从 0 到 n - 1 编号。最初，除 0 号房间外的其余所有房间都被锁住。你的目标是进入所有的房间。然而，你不能在没有获得钥匙的时候进入锁住的房间。</p>
 * <p>当你进入一个房间，你可能会在里面找到一套不同的钥匙，每把钥匙上都有对应的房间号，即表示钥匙可以打开的房间。你可以拿上所有钥匙去解锁其他房间。</p>
 * <p>给你一个数组 rooms 其中 rooms[i] 是你进入 i 号房间可以获得的钥匙集合。如果能进入 所有 房间返回 true，否则返回 false。</p>
 * <p>
 * <b>示例：</b>
 * <pre>
 * 示例 1：
 *      输入：rooms = [[1],[2],[3],[]]
 *      输出：true
 *      解释：
 *          我们从 0 号房间开始，拿到钥匙 1。
 *          之后我们去 1 号房间，拿到钥匙 2。
 *          然后我们去 2 号房间，拿到钥匙 3。
 *          最后我们去了 3 号房间。
 *          由于我们能够进入每个房间，我们返回 true。
 *
 * 示例 2：
 *      输入：rooms = [[1,3],[3,0,1],[2],[0]]
 *      输出：false
 *      解释：我们不能进入 2 号房间。
 * </pre>
 * </p>
 *
 * <p>
 * <b>提示:</b>
 * <ul>
 *     <li>n == rooms.length</li>
 *     <li>2 <= n <= 1000</li>
 *     <li>0 <= rooms[i].length <= 1000</li>
 *     <li>1 <= sum(rooms[i].length) <= 3000</li>
 *     <li>0 <= rooms[i][j] < n</li>
 *     <li>所有 rooms[i] 的值 互不相同</li>
 * </ul>
 * </p>
 *
 * @author c2b
 * @since 2024/2/22 14:19
 */
public class LC0841KeysAndRooms_M {
    static class Solution {

        public boolean canVisitAllRooms(List<List<Integer>> rooms) {
            boolean[] visited = new boolean[rooms.size()];
            dfs1(rooms, 0, visited);
            for (boolean flag : visited) {
                if (!flag) {
                    return false;
                }
            }
            return true;
        }

        private void dfs1(List<List<Integer>> rooms, int currRoom, boolean[] visited) {
            // 如果已访问过当前房间，直接跳过
            if (visited[currRoom]) {
                return;
            }
            // 更新当前房间为已访问过
            visited[currRoom] = true;
            for (Integer target : rooms.get(currRoom)) {
                dfs1(rooms, target, visited);
            }
        }


        public boolean canVisitAllRooms2(List<List<Integer>> rooms) {
            Set<Integer> visitedRooms = new HashSet<>(rooms.size());
            dfs2(rooms, 0, visitedRooms);
            return visitedRooms.size() == rooms.size();
        }

        private void dfs2(List<List<Integer>> rooms, int currRoom, Set<Integer> visitedRooms) {
            if (visitedRooms.contains(currRoom)) {
                return;
            }
            visitedRooms.add(currRoom);
            for (Integer integer : rooms.get(currRoom)) {
                dfs2(rooms, integer, visitedRooms);
            }
        }
    }

    public static void main(String[] args) {
        List<List<Integer>> allRooms1 = new ArrayList<>();

        List<Integer> room1 = new ArrayList<>();
        room1.add(1);
        List<Integer> room2 = new ArrayList<>();
        room2.add(2);
        List<Integer> room3 = new ArrayList<>();
        room3.add(3);
        List<Integer> room4 = new ArrayList<>();

        allRooms1.add(room1);
        allRooms1.add(room2);
        allRooms1.add(room3);
        allRooms1.add(room4);

        List<List<Integer>> allRooms2 = new ArrayList<>();

        List<Integer> roomList1 = new ArrayList<>();
        roomList1.add(1);
        roomList1.add(3);
        List<Integer> roomList2 = new ArrayList<>();
        roomList2.add(3);
        roomList2.add(0);
        roomList2.add(1);
        List<Integer> roomList3 = new ArrayList<>();
        roomList3.add(2);
        List<Integer> roomList4 = new ArrayList<>();
        roomList4.add(0);

        allRooms2.add(roomList1);
        allRooms2.add(roomList2);
        allRooms2.add(roomList3);
        allRooms2.add(roomList4);

        Solution solution = new Solution();

        System.out.println(solution.canVisitAllRooms(allRooms1));
        System.out.println(solution.canVisitAllRooms(allRooms2));
    }
}
